import torch

# 创建一个示例向量
vector = torch.tensor([3.0, 1.0, 4.0, 1.0, 5.0, 9.0, 2.0, 6.0, 5.0, 3.0])

# 要获取的前 N 大的数
N = 3

# 使用 torch.topk 获取前 N 大的数及其索引
values, indices = torch.topk(vector, k=N)

print(f"前 {N} 大的数: {values}")
print(f"对应的索引: {indices}")
